/**
 * 传送带上的包裹必须在 D 天内从一个港口运送到另一个港口。
 * <p>
 * 传送带上的第 i 个包裹的重量为 weights[i]。每一天，我们都会按给出重量的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。
 * <p>
 * 返回能在 D 天内将传送带上的所有包裹送达的船的最低运载能力。
 * 示例 1：
 * <p>
 * 输入：weights = [1,2,3,4,5,6,7,8,9,10], D = 5
 * 输出：15
 * 解释：
 * 船舶最低载重 15 就能够在 5 天内送达所有包裹，如下所示：
 * 第 1 天：1, 2, 3, 4, 5
 * 第 2 天：6, 7
 * 第 3 天：8
 * 第 4 天：9
 * 第 5 天：10
 * <p>
 * 请注意，货物必须按照给定的顺序装运，因此使用载重能力为 14 的船舶并将包装分成 (2, 3, 4, 5), (1, 6, 7), (8), (9), (10) 是不允许的。
 * <p>
 */
class Solution {

    public static void main(String[] args) {
        System.out.println(shipWithinDays(new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}, 1));
    }

    /**
     * 二分法，在结果的最小值和最大值之间二分查找，直到找到
     *
     * @param weights
     * @param d
     * @return
     */
    public static int shipWithinDays(int[] weights, int d) {
        int min = 0, max = 0;
        for (int weight : weights) {
            min = Math.max(weight, min);
            max += weight;
        }
        if (d == 1) {
            return max;
        } else if (d >= weights.length) {
            return min;
        }
        int mid = (min + max) / 2;
        int last = 0;
        do {
            if (match(weights, d, mid)) {
                max = mid;
                last = mid;
            } else {
                min = mid;
            }
            mid = (min + max) / 2;
        } while (mid != min);
        if (match(weights, d, min)) {
            last = min;
        }
        return last;
    }

    public static boolean match(int[] weights, int d, int sum) {
        int count = 0, tmp = 0;
        for (int weight : weights) {
            tmp += weight;
            if (tmp > sum) {
                count++;
                if (count > d) {
                    return false;
                }
                tmp = weight;
            }
        }
        if (tmp > 0) {
            count++;
            return count <= d;
        }
        return true;
    }


}